Python Array Dimensions: Understanding Shape and Size